package com.uhuibao.tmall.framework.web.exception;

import javax.servlet.http.HttpServletRequest;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;

import com.uhuibao.tmall.api.domain.response.ResponseResult;
import com.uhuibao.tmall.common.core.domain.XmlResult;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.validation.BindException;
import org.springframework.web.HttpRequestMethodNotSupportedException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.MissingPathVariableException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;
import com.uhuibao.tmall.common.core.domain.AjaxResult;
import com.uhuibao.tmall.common.exception.DemoModeException;
import com.uhuibao.tmall.common.exception.ServiceException;

import java.io.StringWriter;

/**
 * 全局异常处理器
 *
 * @author ruoyi
 */

/**
 * 全局异常处理器
 *
 * @author ruoyi
 */
@RestControllerAdvice
public class GlobalExceptionHandler {
    private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);

    /**
     * 权限校验异常
     */
    @ExceptionHandler(AccessDeniedException.class)
    public ResponseEntity<String> handleAccessDeniedException(AccessDeniedException e, HttpServletRequest request) {
        String requestURI = request.getRequestURI();
        log.error("请求地址'{}',权限校验失败'{}'", requestURI, e.getMessage());
//            return AjaxResult.error(HttpStatus.FORBIDDEN, "没有权限，请联系管理员授权");

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc("没有权限，请联系管理员授权");

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 请求方式不支持
     */
    @ExceptionHandler(HttpRequestMethodNotSupportedException.class)
    public ResponseEntity<String> handleHttpRequestMethodNotSupported(HttpRequestMethodNotSupportedException e,
                                                                      HttpServletRequest request) {
        String requestURI = request.getRequestURI();
        log.error("请求地址'{}',不支持'{}'请求", requestURI, e.getMethod());
//            return AjaxResult.error(e.getMessage());

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc(e.getMessage());

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 业务异常
     */
    @ExceptionHandler(ServiceException.class)
    public ResponseEntity<String> handleServiceException(ServiceException e, HttpServletRequest request) {
        log.error(e.getMessage(), e);
        Integer code = e.getCode();
//            return StringUtils.isNotNull(code) ? AjaxResult.error(code, e.getMessage()) : AjaxResult.error(e.getMessage());

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode(e.getMessage());
        responseResult.setErrDesc(e.getDetailMessage());

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 请求路径中缺少必需的路径变量
     */
    @ExceptionHandler(MissingPathVariableException.class)
    public ResponseEntity<String> handleMissingPathVariableException(MissingPathVariableException e, HttpServletRequest request) {
        String requestURI = request.getRequestURI();
        log.error("请求路径中缺少必需的路径变量'{}',发生系统异常.", requestURI, e);
//            return AjaxResult.error(String.format("请求路径中缺少必需的路径变量[%s]", e.getVariableName()));

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc("发生系统异常");

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 请求参数类型不匹配
     */
    @ExceptionHandler(MethodArgumentTypeMismatchException.class)
    public ResponseEntity<String> handleMethodArgumentTypeMismatchException(MethodArgumentTypeMismatchException e, HttpServletRequest request) {
        String requestURI = request.getRequestURI();
        log.error("请求参数类型不匹配'{}',发生系统异常.", requestURI, e);
//            return AjaxResult.error(String.format("请求参数类型不匹配，参数[%s]要求类型为：'%s'，但输入值为：'%s'", e.getName(), e.getRequiredType().getName(), e.getValue()));

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc("请求参数类型不匹配");

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 拦截未知的运行时异常
     */
    @ExceptionHandler(RuntimeException.class)
    public ResponseEntity<String> handleRuntimeException(RuntimeException e, HttpServletRequest request) {
        String requestURI = request.getRequestURI();
        log.error("请求地址'{}',发生未知异常.", requestURI, e);
//            return AjaxResult.error(e.getMessage());

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc("发生未知异常");

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 系统异常
     */
    @ExceptionHandler(Exception.class)
    public ResponseEntity<String> handleException(Exception e, HttpServletRequest request) {
        String requestURI = request.getRequestURI();
        log.error("请求地址'{}',发生系统异常.", requestURI, e);
//            return AjaxResult.error(e.getMessage());

        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc("发生系统异常");

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 自定义验证异常
     */
    @ExceptionHandler(BindException.class)
    public ResponseEntity<String> handleBindException(BindException e, HttpServletRequest request) {
        log.error(e.getMessage(), e);
        String message = e.getAllErrors().get(0).getDefaultMessage();
//            return AjaxResult.error(message);

        // 获取请求参数
        String tbOrderNoParam = request.getParameter("tbOrderNo");
        Long tbOrderNo = null;

        String bizType = request.getParameter("bizType");
        ResponseResult responseResult = new ResponseResult();
        if (tbOrderNoParam != null && !tbOrderNoParam.isEmpty()) {
            tbOrderNo = Long.valueOf(tbOrderNoParam);
            // 执行需要 tbOrderNo 的逻辑
            responseResult.setTbOrderNo(tbOrderNo);
        }
        if (bizType != null && !bizType.isEmpty()) {
            responseResult.setBizType(bizType);
        }

        responseResult.setErrCode("MISSING_PARAMETERS");
        responseResult.setErrDesc("参数缺失");


        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    /**
     * 自定义验证异常
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ResponseEntity<String> handleMethodArgumentNotValidException(MethodArgumentNotValidException e) {
        log.error(e.getMessage(), e);
        String message = e.getBindingResult().getFieldError().getDefaultMessage();
//            return AjaxResult.error(message);


        ResponseResult responseResult = new ResponseResult();
        responseResult.setErrCode("500");
        responseResult.setErrDesc(message);

        String xmlResponse = convertObjectToXml(responseResult);

        return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                .contentType(MediaType.valueOf(MediaType.APPLICATION_XML + ";charset=GBK"))
                .body(xmlResponse);
    }

    private String convertObjectToXml(Object object) {
        try {
            JAXBContext jaxbContext = JAXBContext.newInstance(object.getClass());
            Marshaller marshaller = jaxbContext.createMarshaller();
            marshaller.setProperty(Marshaller.JAXB_ENCODING, "GBK");
            marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);

            StringWriter stringWriter = new StringWriter();
            stringWriter.write("<?xml version=\"1.0\" encoding=\"GBK\"?>\n");
            marshaller.marshal(object, stringWriter);

            return stringWriter.toString();
        } catch (JAXBException e) {
            e.printStackTrace();
            return null;
        }
    }
}
